package com.ittao.taoaicodecommon.config;

import com.qcloud.cos.COSClient;
import com.qcloud.cos.ClientConfig;
import com.qcloud.cos.auth.BasicCOSCredentials;
import com.qcloud.cos.http.HttpProtocol;
import com.qcloud.cos.region.Region;
import lombok.Data;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
@ConfigurationProperties(prefix = "cos.client")
@ConditionalOnProperty(
        prefix = "cos.client",
        name = {"host", "secretId", "secretKey", "region", "bucket"}
)
@Data
public class CosClientConfig {

    /**
     * 域名
     */
    private String host;

    /**
     * secretId
     */
    private String secretId;

    /**
     * secretKey 密钥
     */
    private String secretKey;

    /**
     * 区域
     */
    private String region;

    /**
     * 存储桶名称
     */
    private String bucket;

    @Bean
    public COSClient cosClient() {
        // 创建基本的COS凭证
        BasicCOSCredentials cred = new BasicCOSCredentials(secretId, secretKey);
        // 设置存储桶所在的区域
        ClientConfig clientConfig = new ClientConfig(new Region(region));
        // 设置请求协议
        clientConfig.setHttpProtocol(HttpProtocol.https);

        // 性能优化配置
        // 1. 超时设置优化
        clientConfig.setSocketTimeout(60000);           // 60秒 - 适合大文件下载
        clientConfig.setConnectionTimeout(10000);       // 10秒 - 连接建立超时
        clientConfig.setConnectionRequestTimeout(5000); // 5秒 - 从连接池获取连接超时

        // 2. 连接池优化
        clientConfig.setMaxConnectionsCount(100);       // 增加连接池大小到100
        clientConfig.setMaxErrorRetry(3);              // 设置重试次数为3次

        // 3. 性能调优
        clientConfig.setUserAgent("tao-ai-code-mother/1.0"); // 自定义User-Agent
        clientConfig.setUseBasicAuth(false);           // 禁用基础认证以提升性能

        // 返回一个COSClient实例
        return new COSClient(cred, clientConfig);
    }
}
